ball animation using python